package com.c2b.algorithm.leetcode.base;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * <a href='https://leetcode.cn/problems/longest-harmonious-subsequence/'>最长和谐子序列(Longest Harmonious Subsequence)</a>
 * <p>和谐数组是指一个数组里元素的最大值和最小值之间的差别 正好是 1 。</p>
 * <p>现在，给你一个整数数组 nums ，请你在所有可能的子序列中找到最长的和谐子序列的长度。</p>
 * <p>数组的子序列是一个由数组派生出来的序列，它可以通过删除一些元素或不删除元素、且不改变其余元素的顺序而得到。</p>
 *
 * <p>
 * <b>示例：</b>
 * <pre>
 * 示例 1：
 *      输入：nums = [1,3,2,2,5,2,3,7]
 *      输出：5
 *      解释：最长的和谐子序列是 [3,2,2,2,3]
 *
 * 示例 2：
 *      输入：nums = [1,2,3,4]
 *      输出：2
 *
 * 示例 3：
 *      输入：nums = [1,1,1,1]
 *      输出：0
 * </pre>
 * </p>
 * <p>
 * <b>提示：</b>
 *     <ul>
 *         <li>1 <= nums.length <= 2 * 10^4</li>
 *         <li>-10^9 <= nums[i] <= 10^9</li>
 *     </ul>
 * </p>
 *
 * @author c2b
 * @since 2023/12/18 11:15
 */
public class LC0594LongestHarmoniousSubsequence_S {
    static class Solution {

        public int findLHS1(int[] nums) {
            Arrays.sort(nums);
            int ans = 0;
            for (int begin = 0, end = 0; end < nums.length; end++) {
                // 如果 end位置的值 比 begin位置的值，差值大于1；将 begin 向左移
                while (nums[end] - nums[begin] > 1) {
                    begin++;
                }
                // 如果 end位置的值 比 begin位置的值，差值等于1；汇总并将 end 向右移
                // 如果 end位置的值 比 begin位置的值，差值小于1；无需汇总直接将 end 向右移
                if (nums[end] - nums[begin] == 1) {
                    ans = Math.max(ans, end - begin + 1);
                }
            }
            return ans;
        }

        public int findLHS(int[] nums) {
            Map<Integer, Integer> cntMap = new HashMap<>();
            for (int num : nums) {
                cntMap.put(num, cntMap.getOrDefault(num, 0) + 1);
            }
            int ans = 0;
            for (Map.Entry<Integer, Integer> entry : cntMap.entrySet()) {
                if (cntMap.containsKey(entry.getKey() - 1)) {
                    ans = Math.max(ans, entry.getValue() + cntMap.get(entry.getKey() - 1));
                }
            }
            return ans;
        }
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        System.out.println(solution.findLHS(new int[]{1, 3, 2, 2, 5, 2, 3, 7}));
        System.out.println(solution.findLHS(new int[]{1,2,3,4}));
        System.out.println(solution.findLHS(new int[]{1,1,1,1}));
    }
}
